@echo off
chcp 65001 >nul
echo ========================================
echo 显卡测试工具打包脚本（修复OCR初始化）
echo ========================================
echo.

:: 核心修改：添加PaddleOCR模型文件和依赖收集
pyinstaller --noconfirm --windowed ^
--name "测试助手" ^
--icon "icon.ico" ^
--add-data "test_results.db;." ^
--add-data ".venv\Lib\site-packages\paddlex\.version;paddlex" ^
--add-data ".venv\Lib\site-packages\paddleocr;paddleocr" ^
--additional-hooks-dir=. ^
--hidden-import "paddleocr" ^
--hidden-import "paddle" ^
--hidden-import "paddlex" ^
--hidden-import "pyclipper" ^
--hidden-import "shapely" ^
--hidden-import "opencv-python" ^
--hidden-import "PIL" ^
--hidden-import "skimage" ^
--hidden-import "PIL.Image" ^
--hidden-import "keyboard" ^
--hidden-import "sqlite3" ^
--collect-data "paddleocr" ^
--collect-submodules "paddleocr" ^
--collect-data "paddlex" ^
main.py

if errorlevel 1 (
    echo.
    echo 错误: 打包失败！
    echo 请检查错误信息并重试
    pause
    exit /b 1
)

:: 清理临时文件
rmdir /s /q build >nul 2>&1
rmdir /s /q __pycache__ >nul 2>&1
del /q *.spec >nul 2>&1

echo.
echo ========================================
echo 打包成功！（文件夹模式）
echo ========================================
echo.
echo 可执行文件位置: dist\测试助手\测试助手.exe
echo 所需的依赖文件已包含在 dist\测试助手 目录中
echo.
echo 提示：PaddleOCR 需要额外的模型文件，请确保以下目录存在：
echo      dist\测试助手\paddleocr
echo      dist\测试助手\paddlex
echo.
echo 打包完成！可以运行 dist\测试助手\测试助手.exe 了
echo.
pause